package cn.com.code.design.model.aggregates;

import cn.com.code.design.model.vo.TreeNode;
import cn.com.code.design.model.vo.TreeRoot;
import java.util.Map;

/**
 * 规则树聚合
 * @author zhengpei
 * @date 2021/9/27
 */
public class TreeRich {

  private TreeRoot treeRoot;                          //树根信息
  private Map<Long, TreeNode> treeNodeMap;        //树节点ID -> 子节点

  public TreeRich(TreeRoot treeRoot, Map<Long, TreeNode> treeNodeMap) {
    this.treeRoot = treeRoot;
    this.treeNodeMap = treeNodeMap;
  }

  public TreeRoot getTreeRoot() {
    return treeRoot;
  }

  public void setTreeRoot(TreeRoot treeRoot) {
    this.treeRoot = treeRoot;
  }

  public Map<Long, TreeNode> getTreeNodeMap() {
    return treeNodeMap;
  }

  public void setTreeNodeMap(Map<Long, TreeNode> treeNodeMap) {
    this.treeNodeMap = treeNodeMap;
  }

}
